#写一个目录统计字节数的函数（目录中带子目录）
import os
#定义统计目录文件的函数
def dir_dir(d):
    #初始值设定为0
    num=0
    #判断是否是文件
    if os.path.isfile(d):
        #对里面的文件进行累加操作
        num+=os.path.getsize(d)
    #判断是否为目录
    if os.path.isdir(d):
        #统计目录里的所有文件
        dir_list=os.listdir(d)
        # 遍历目录里的文件
        for i in dir_list:
            #对路径进行添加
            b=os.path.join(d,i)
            #判断是否文件
            if os.path.isfile(b):
                num+=os.path.getsize(b)
            #判断是否目录
            if os.path.isdir(b):
                num+=dir_dir(b)#对目录递归统计
    return num
#设置操作函数运行前提条件
while True:
    key_world=input('请输入要统计的路径（例如：C:\\Users\\Administrator\\Desktop\\ceshi,或按enter键退出:\n')
    #当输入的不是路径的时候退出
    if key_world == "":
        break
    #当输入的事有效路径执行函数
    else:
        print("共：{} 字节".format(dir_dir(key_world)))

